package cn.tedu.boot07.controller;

import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.multipart.MultipartFile;

import java.io.File;
import java.io.IOException;
import java.util.UUID;

@RestController
public class UploadController {
    @RequestMapping("/upload")
    public String upload(MultipartFile picFile) throws IOException {
        //得到原始文件的名称   a.jpg
        String fileName= picFile.getOriginalFilename();
        System.out.println("文件名:"+fileName);
        //得到文件的类型   .jpg
        String suffix = fileName.substring(fileName.lastIndexOf("."));
        System.out.println(suffix);
        //得到唯一的文件名  UUID.randomUUID()得到唯一标识符
        fileName = UUID.randomUUID()+suffix;
        System.out.println(fileName);
        //得到保存图片的文件夹路径
        String dirPath = "d:/files";
        File dirFile = new File(dirPath);
        //判断如果文件不存在
        if (!dirFile.exists()){
            //创建文件夹
            dirFile.mkdirs();
        }
        //得到完整的文件路径
        String filePath = dirPath+"/"+fileName;
        //把文件保存到上面的路径  异常抛出
        picFile.transferTo(new File(filePath));
        System.out.println("保存完成!");
        //xxx.jpg
        //    /xxx.jpg
        return fileName;

    }


    @RequestMapping("/remove")
    public void remove(String fileName){

        System.out.println("fileName = " + fileName);
        //删除文件
        new File("d:/files/"+fileName).delete();
    }
}
